import torch

# 创建一个一维张量
tensor_1d = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float32)
mean_1 = torch.mean(tensor_1d)
mean_2 = tensor_1d.mean()
print("一维张量的平均值:", mean_1)
print("一维张量的平均值:", mean_2)

# 创建一个二维张量
tensor_2d = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
mean_all = torch.mean(tensor_2d)
print("二维张量的整体平均值:", mean_all)

# 按行计算平均值（dim=1）
mean_row = torch.mean(tensor_2d, dim=1)
print("按行计算的平均值:", mean_row)

# 按列计算平均值（dim=0）
mean_col = torch.mean(tensor_2d, dim=0)
print("按列计算的平均值:", mean_col)